bs=8
gpus=8
num_epochs=3000
output_dir=experiments/multitask_3_output
path_to_ImageNetC=/ifs/root/ipa01/101/user_101002/Project/ControlVAR-main/dataset/gopro
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 # 指定要使用的GPU


python3 -m torch.distributed.launch --nproc_per_node=8 --master_port=29500 train_control_var_hpu.py \
--batch_size $bs \
--dataset_name multi_task \
--num_epochs $num_epochs \
--data_dir $path_to_ImageNetC \
--gpus $gpus \
--output_dir $output_dir \
--multi_cond True \
--config configs/train_mask_var_imagenet2012_gray_d16.yaml \
--var_pretrained_path /ifs/root/ipa01/101/user_101002/Project/ControlVAR-main/pretrained/var_d16.pth


# CUDA_VISIBLE_DEVICES=0 python3 train_control_var_hpu.py \
# --batch_size $bs \
# --dataset_name imagenetC_gray \
# --data_dir $path_to_ImageNetC \
# --gpus $gpus \
# --output_dir $output_dir \
# --multi_cond True \
# --config configs/train_mask_var_imagenet2012_gray_d12.yaml \
# --var_pretrained_path /ifs/root/ipa01/101/user_101002/Project/ControlVAR-main/pretrained/var_d16.pth